cd ../..

gpus=$1

for ood_size in 96
do
    python semi_mae_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_joint.yaml \
        --exp_name semi/BYOLA_3C+R/frame_olBYOLA_0.3MAE_MASKFrameloss+SupMT/o${ood_size}_i24 \
        --feat_dim 256 \
        --freeze_cnn_layer "[0, 1, 2]" \
        --self_target "crnn" \
        --batch_sizes "[0, 12, 12, 24, ${ood_size}]" \
        --joint_itpl_warmup 50 \
        --joint_itpl_exponent 5 \
        --rmc_time_scale 1.0 \
        --rmc_freq_scale '[1.0, 1.0]' \
        --rmc_crop_size '[1.0, 1.0]' \
        --test_from_checkpoint "/home/shaonian/SED/sssl_sed/codes/exp/semi/BYOLA_3C+R/frame_olBYOLA_0.5MAE_MASKFrameloss/o96_i24/version_0/epoch=169-obj_metric=1.284.ckpt"

        # --mask_rate 0.3 \
         
        # --nb_filters '[16, 32, 64, 128, 256, 256, 256]' 
done   